#coding=utf-8

"""

目录名	说明
/captcha/	图片验证码生成
commons.py	通用设施文件(正则url、登录验证装饰器)
image_storage.py	云存储设施文件(七牛云)
response_code.py	自定义状态码
sms.py	发送短信
"""
import functools
from flask import g,session,jsonify
from werkzeug.routing import BaseConverter
from ihome.utils.response_code import RET


class RegexConverter(BaseConverter):
    """自定义正则转换器
    在路由中使用正则表达式进行提取参数的转换工具"""
    def __init__(self,url_map,*args):
        super(RegexConverter,self).__init__(url_map)
        self.regex = args[0]


def login_required(f):
    """要求用户登录的验证装饰器"""
    @functools.wraps(f)
    def wrapper(*args,**kwargs):
        user_id = session.get('user_id')
        if user_id is None:
            return jsonify(errno = RET.SERVERERR,errmsg ='用户未登录')
        else:
            g.user_id = user_id
            return f(*args,**kwargs)
    return wrapper











